from python_ai.category.data.mnist_dataset import x_train, x_test, x_val, y_train, y_test, y_val, shape_

SPE_NUM = 5
y_train = (y_train == SPE_NUM).astype(int)
y_test = (y_test == SPE_NUM).astype(int)
y_val = (y_val == SPE_NUM).astype(int)

print('positive percent:', (y_train.sum() + y_val.sum() + y_test.sum()) / (len(y_train) + len(y_val) + len(y_test)))
